package com.ruoyi.energy.service.impl;

import cn.hutool.core.lang.tree.Tree;
import cn.hutool.core.lang.tree.TreeNodeConfig;
import cn.hutool.core.lang.tree.TreeUtil;
import cn.hutool.core.util.IdUtil;
import com.ruoyi.common.core.utils.DateUtils;
import com.ruoyi.common.core.utils.SecurityUtils;
import com.ruoyi.common.core.utils.StringUtils;
import com.ruoyi.common.core.web.domain.AjaxResult;
import com.ruoyi.energy.domain.QyEnterpriseType;
import com.ruoyi.energy.mapper.QyEnterpriseTypeMapper;
import com.ruoyi.energy.service.IQyEnterpriseTypeService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.List;

/**
 * 企业类型Service业务层处理
 *
 * @author cjie
 * @date 2021-11-06
 */
@Service
public class QyEnterpriseTypeServiceImpl implements IQyEnterpriseTypeService {
    @Autowired
    private QyEnterpriseTypeMapper qyEnterpriseTypeMapper;

    /**
     * 查询企业类型
     *
     * @param id 企业类型ID
     * @return 企业类型
     */
    @Override
    public QyEnterpriseType selectQyEnterpriseTypeById(String id) {
        return qyEnterpriseTypeMapper.selectQyEnterpriseTypeById(id);
    }

    /**
     * 查询企业类型列表
     *
     * @param qyEnterpriseType 企业类型
     * @return 企业类型
     */
    @Override
    public List<QyEnterpriseType> selectQyEnterpriseTypeList(QyEnterpriseType qyEnterpriseType) {
        return qyEnterpriseTypeMapper.selectQyEnterpriseTypeList(qyEnterpriseType);
    }

    /**
     * 新增企业类型
     *
     * @param qyEnterpriseType 企业类型
     * @return 结果
     */
    @Override
    public int insertQyEnterpriseType(QyEnterpriseType qyEnterpriseType) {
        qyEnterpriseType.setId(IdUtil.getSnowflake(0, 0).nextIdStr());
        qyEnterpriseType.setCreateTime(DateUtils.getNowDate());
        qyEnterpriseType.setCreateBy(SecurityUtils.getUsername());
        return qyEnterpriseTypeMapper.insertQyEnterpriseType(qyEnterpriseType);
    }

    /**
     * 修改企业类型
     *
     * @param qyEnterpriseType 企业类型
     * @return 结果
     */
    @Override
    public int updateQyEnterpriseType(QyEnterpriseType qyEnterpriseType) {
        qyEnterpriseType.setUpdateTime(DateUtils.getNowDate());
        qyEnterpriseType.setUpdateBy(SecurityUtils.getUsername());
        return qyEnterpriseTypeMapper.updateQyEnterpriseType(qyEnterpriseType);
    }

    /**
     * 批量删除企业类型
     *
     * @param ids 需要删除的企业类型ID
     * @return 结果
     */
    @Override
    public int deleteQyEnterpriseTypeByIds(String[] ids) {
        return qyEnterpriseTypeMapper.deleteQyEnterpriseTypeByIds(ids);
    }

    /**
     * 删除企业类型信息
     *
     * @param id 企业类型ID
     * @return 结果
     */
    @Override
    public AjaxResult deleteQyEnterpriseTypeById(String id) {
        if(hasChildById(id)){
            return AjaxResult.error("存在下级,不允许删除");
        }

        if (checkTypeIsUsed(id)) {
            return AjaxResult.error("类型存在公司,不允许删除");
        }

        return qyEnterpriseTypeMapper.deleteQyEnterpriseTypeById(id) > 0 ? AjaxResult.success() : AjaxResult.error();
    }

    @Override
    public AjaxResult selectTree(QyEnterpriseType qyEnterpriseType) {
        List<QyEnterpriseType> qyEnterpriseTypes = qyEnterpriseTypeMapper.selectQyEnterpriseTypeList(qyEnterpriseType);
        if(StringUtils.isNotEmpty(qyEnterpriseType.getExcludeId())){
            qyEnterpriseTypes.removeIf(type -> qyEnterpriseType.getExcludeId().equals(type.getId()));
        }
        TreeNodeConfig treeNodeConfig = new TreeNodeConfig();

        List<Tree<String>> treeNodes = TreeUtil.build(qyEnterpriseTypes, "0", treeNodeConfig,
                (treeNode, tree) -> {
                    tree.setId(treeNode.getId());
                    tree.setParentId(treeNode.getParentId());
                    tree.setName(treeNode.getEnterpriseType());

                    tree.putExtra("feeAmount", treeNode.getFeeAmount());
                });
        return AjaxResult.success(treeNodes);
    }

    @Override
    public boolean hasChildById(String id) {
        int result = qyEnterpriseTypeMapper.hasChildById(id);
        return result > 0 ? true : false;
    }

    @Override
    public boolean checkTypeIsUsed(String id) {
        return qyEnterpriseTypeMapper.checkTypeIsUsed(id) > 0 ? true : false;
    }
}
